from .vit_moe import VisionTransformer as vmoe
from .vit_ease import VisionTransformer as vease
from .vit_baseSingle import VisionTransformer as vbase_single
from .vit_baseMulti import VisionTransformer as vbase_multi
from .vit_ssiat import VisionTransformer as vssiat

"""
Model Briefing

- vmoe: Single-feature Vision Transformer with MoE
    - inherit from models.components.backbone.vit
    - forward_feats: used for training and test stage, utilize all the additional modules of all tasks
- vease: Single-feature Vision Transformer with EASE
    - TODO: remove the useless backbone
- vbase_single: Single-feature Vision Transformer with one additional module for all tasks
    - inherit from models.components.backbone.vit
- vbase_multi: Single-feature Vision Transformer with one additional module for each task
    - inherit from models.components.backbone.vit
- vssiat: Single-feature Vision Transformer with SSIAT
    - inherit from models.components.backbone.vit
    - forward_feats: one additional module for all tasks
"""

__all__ = [
    "vmoe",
    "vease",
    "vbase_single",
    "vbase_multi",
    "vssiat",
]
